"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
const config_1 = require("src/config");
const index_1 = require("../manager/index");
function AddToQueue(config) {
    return (target, key, descriptor) => {
        const ins = config_1.getTaskInstance(target.constructor);
        const originFn = descriptor.value;
        descriptor.value = (...args) => __awaiter(this, void 0, void 0, function* () {
            let result = yield originFn.apply(ins, args);
            if (!!result) {
                result = result.constructor == Array ? result : [result];
                index_1.queueManager.addToQueue({
                    queueName: config.name,
                    urls: result
                });
            }
            return result;
        });
        return descriptor;
    };
}
exports.default = AddToQueue;
